package com.edoomath.api.utils;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.List;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ZipFileUtils {
	
	private static final Logger logger = LoggerFactory.getLogger(ZipFileUtils.class);
	
	static final int BUFFER = 8192;
	
    private File zipFile;
    
    /**
     * 
     * @param pathName
     */
    public ZipFileUtils(String pathName) {  
        zipFile = new File(pathName);  
    }
    
    /**
     * 
     * @param pathName
     */
    public void compress(List<String> pathName) {
        ZipOutputStream out = null;  
        try { 
            FileOutputStream fileOutputStream = new FileOutputStream(zipFile);  
            CheckedOutputStream cos = new CheckedOutputStream(fileOutputStream,  
                    new CRC32());  
            out = new ZipOutputStream(cos);  
            String basedir = "";
            for (int i=0; i<pathName.size(); i++ ){
                compress(new File(pathName.get(i)), out, basedir);  
            }
            out.close(); 
        } catch (Exception e) {  
        	logger.error("error compress file", e);
            throw new RuntimeException(e);  
        }
    }  
    
    /**
     * 
     * @param srcPathName
     */
    public void compress(String srcPathName) {  
        File file = new File(srcPathName);  
        if (!file.exists())  
            throw new RuntimeException(srcPathName + "不存在！");  
        try {  
            FileOutputStream fileOutputStream = new FileOutputStream(zipFile);  
            CheckedOutputStream cos = new CheckedOutputStream(fileOutputStream,  
                    new CRC32());  
            ZipOutputStream out = new ZipOutputStream(cos);  
            String basedir = "";  
            compress(file, out, basedir);  
            out.close();  
        } catch (Exception e) {
        	logger.error("error compress file", e);
            throw new RuntimeException(e);  
        }  
    }  
    
    /**
     * 
     * @param file
     * @param out
     * @param basedir
     */
    private void compress(File file, ZipOutputStream out, String basedir) {  
        /* 判断是目录还是文件 */ 
        if (file.isDirectory()) { 
        	logger.info("压缩：" + basedir + file.getName());
            this.compressDirectory(file, out, basedir);  
        } else {  
        	logger.info("压缩：" + basedir + file.getName());  
            this.compressFile(file, out, basedir);  
        }  
    }  
   
    /**
     * 
     * @param dir
     * @param out
     * @param basedir
     */
    private void compressDirectory(File dir, ZipOutputStream out, String basedir) {  
        if (!dir.exists())  
            return;  
   
        File[] files = dir.listFiles();  
        for (int i = 0; i < files.length; i++) {  
            /* 递归 */ 
            compress(files[i], out, basedir + dir.getName() + "/");  
        }  
    }  
   
    /**
     * 
     * @param file
     * @param out
     * @param basedir
     */
    private void compressFile(File file, ZipOutputStream out, String basedir) {  
        if (!file.exists()) {  
            return;  
        }  
        try {  
            BufferedInputStream bis = new BufferedInputStream(  
                    new FileInputStream(file));  
            ZipEntry entry = new ZipEntry(basedir + file.getName());  
            out.putNextEntry(entry);  
            int count;  
            byte data[] = new byte[BUFFER];  
            while ((count = bis.read(data, 0, BUFFER)) != -1) {  
                out.write(data, 0, count);  
            }  
            bis.close();  
        } catch (Exception e) {  
        	logger.error("error compress file", e);
            throw new RuntimeException(e);  
        }  
    }
    
//   public static void main(String[] args) {  
//        ZipCompressor zc = new ZipCompressor("C:/Users/wj/Desktop/aaa.zip");  
//        zc.compress("C:/Users/wj/Desktop/water.js","C:/Users/wj/Desktop/index.jsp"); 
//    }

}
